package com.project.controller;

import com.baomidou.mybatisplus.core.metadata.IPage;
import com.project.bean.UserBean;
import com.project.service.IUserService;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;

import javax.servlet.http.HttpServletRequest;

@RestController
@RequestMapping("user")
public class UserController {
    @Autowired
    private IUserService userService;

    @RequestMapping("login")
    public String login(String userName, String pwd, HttpServletRequest req) {
        UserBean user = userService.login(userName, pwd);
        if (user != null) {
            req.getSession().setAttribute("loginUser", user);
            return "ok";
        }
        return "no";
    }

    @RequestMapping("getLoginUser")
    public UserBean getLoginUser(HttpServletRequest req) {
        return (UserBean) req.getSession().getAttribute("loginUser");
    }

    @RequestMapping("updatePwd")
    public String updatePwd(Integer userId, String pwd) {
        userService.updatePwd(userId, pwd);
        return "ok";
    }

    @RequestMapping("findByItem")
    public IPage<UserBean> findByItem(Integer pageNo, String userName) {
        return userService.findByItem(pageNo, userName);
    }

    @RequestMapping("add")
    public String add(UserBean user, Integer[] roleId) {
        if (userService.add(user, roleId)) {
            return "ok";
        }
        return "no";
    }

    @RequestMapping("del")
    public String del(Integer id, HttpServletRequest req) {
        UserBean loginUser = (UserBean) req.getSession().getAttribute("loginUser");
        if (userService.del(id, loginUser)) {
            return "ok";
        }
        return "no";
    }

    @RequestMapping("findById")
    public UserBean findById(Integer id) {
        return userService.findById(id);
    }

    @RequestMapping("updateRole")
    public String updateRoles(Integer userId, Integer[] roleId) {
        userService.updateRole(userId, roleId);
        return "ok";
    }
}
